import sqlite3
import os
import sys
from pathlib import Path

# 添加项目根目录到 Python 路径
sys.path.insert(0, os.path.abspath(os.path.join(os.path.dirname(__file__), '..')))

# 获取数据库路径
ROOT_DIR = Path(__file__).parent.parent
DB_PATH = ROOT_DIR / "articles.db"

def recreate_prompt_templates_table():
    """重新创建提示词模板表"""
    conn = sqlite3.connect(str(DB_PATH))
    cursor = conn.cursor()
    
    # 删除现有表（如果存在）
    cursor.execute("DROP TABLE IF EXISTS prompt_templates")
    
    # 创建新表
    cursor.execute("""
        CREATE TABLE IF NOT EXISTS prompt_templates (
            id TEXT PRIMARY KEY,
            name TEXT NOT NULL,
            description TEXT,
            system_prompt TEXT,
            user_prompt TEXT NOT NULL,
            variables TEXT,  -- JSON 格式存储变量列表
            category TEXT,
            tags TEXT,  -- JSON 格式存储标签列表
            created_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP,
            updated_at TIMESTAMP DEFAULT CURRENT_TIMESTAMP
        )
    """)
    
    conn.commit()
    conn.close()
    
    print(f"提示词模板表已重新创建: {DB_PATH}")

if __name__ == "__main__":
    recreate_prompt_templates_table() 